(PYTHON) Day - 1 Introduction

파이썬을 파이썬답게!

Reference

  • 문제 출처 - HackerRank
  • 파이썬 연습 - Practice - Python

Python 기초를 탄탄히 다지자
문제를 풀고 Discussion Tab을 참고하며 코드 스타일을 개선하려고 노력하고자 합니다


Introduction

loops

N 개의 정수를 입력 받고, 그 정수들의 제곱을 출력하는 문제
정수마다 한 줄씩 띄워야함!

n = int(input())

ans = [i ** 2 for i in range(n)]
for idx in range(n):
print(ans[idx])

언패킹을 사용해서 한 줄로 코딩 가능!

print(\*[num**2 for num in range(n)], sep='\n') 

혹은 리스트에 담아서 출력하면 됨

[print(i**2) for i in range(n)] 

if-else

홀수 이거나, 6 에서 20 사이의 숫자이면 ‘Weird’를 출력
그 외에는 ‘Not Weird’ 출력

n = int(input().strip())

if n % 2 == 0:
if 2 <= n <= 5:
print('Not Weird')
elif 6 <= n <= 20:
print('Weird')
else:
print('Not Weird')
else:
print('Weird')

dict 자료형을 사용하면 if-else문을 사용하지 않고도 표현 가능

n = int(input().strip())
check = {True: 'Not Weird', False: 'Weird'}

print(check[n % 2 == 0 and (n in range(2, 6) or n > 20)])